import os
os.environ["OPENAI_API_KEY"] = "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"
os.environ["OPENAI_API_BASE"] = "http://107.148.33.225/v1"

from langchain.agents import create_csv_agent
from langchain.llms import OpenAI

agent = create_csv_agent(OpenAI(temperature=0.1), '../source/wuzidata.csv', verbose=True)

res = agent.run("有哪些料仓?")
print(res)

# agent.run("有哪些设备?")
# agent.run("1号仓有哪些材料?")
# agent.run("1号仓2019年11月22日0dia的详细数据?")
